package club.xiaojiawei.dp;

/**
 * @author 肖嘉威
 * @version 1.0
 * @date 6/3/22 8:07 PM
 * @question 474. 一和零
 * @description 给你一个二进制字符串数组 strs 和两个整数 m 和 n 。
 * 请你找出并返回 strs 的最大子集的长度，该子集中 最多 有 m 个 0 和 n 个 1 。
 * 如果 x 的所有元素也是 y 的元素，集合 x 是集合 y 的 子集 。
 */
public class FindMaxForm474 {

    public static void main(String[] args) {
        FindMaxForm474 test = new FindMaxForm474();
        int result = test.findMaxForm(new String[]{"10", "0", "1"}, 1, 1);
        System.out.println("最大长度："+result);
    }

    /**
     * dp（0/1背包）
     * 时间复杂度：O(lmn+L)
     * 空间复杂度：O(mn + 2l)
     * @param strs
     * @param m
     * @param n
     * @return
     */
    public int findMaxForm(String[] strs, int m, int n) {
        int length = strs.length;
        int[][] arr = new int[length][2];
        for (int i = 0; i < length; i++) {
            arr[i][0] = strs[i].length() - (arr[i][1] = strs[i].replace("0","").length());
        }
        int[][] dp = new int[m + 1][n + 1];
        int w, h;
        for (int i = 0; i < length; i++) {
            w = arr[i][0];
            h = arr[i][1];
            for (int x = m; x >= w && n >= h; x--){
                for (int y = n; y >= h; y--){
                    dp[x][y] = Math.max(dp[x][y], dp[x - w][y - h] + 1);
                }
            }
        }
        return dp[m][n];
    }

    /**
     * 官方-dp
     * @param strs
     * @param m
     * @param n
     * @return
     */
    public int findMaxForm2(String[] strs, int m, int n) {
        int[][] dp = new int[m + 1][n + 1];
        for (String str : strs) {
            int[] zerosOnes = getZerosOnes(str);
            int zeros = zerosOnes[0], ones = zerosOnes[1];
            for (int j = m; j >= zeros; j--) {
                for (int k = n; k >= ones; k--) {
                    dp[j][k] = Math.max(dp[j][k], dp[j - zeros][k - ones] + 1);
                }
            }
        }
        return dp[m][n];
    }

    public int[] getZerosOnes(String str) {
        int[] zerosOnes = new int[2];
        int length = str.length();
        for (int i = 0; i < length; i++) {
            zerosOnes[str.charAt(i) - '0']++;
        }
        return zerosOnes;
    }
}
